Skip to content

Conversation

@ryanmrichard
Copy link
Member

Is this pull request associated with an issue(s)?
No.

Description
For implementing DFT I need mixed contractions (combinations of element-wise and normal contraction). This PR will add them. I'm also starting some general machinery for dealing with Einsum though I'm going to stop short of implementing it (not going to do the traces).

TODOs

  • Implement batched contraction.

@ryanmrichard ryanmrichard self-assigned this Sep 24, 2025
@ryanmrichard ryanmrichard marked this pull request as ready for review September 25, 2025 20:45
@ryanmrichard
Copy link
Member Author

I'm going to punt on the rest of this for now. It requires us to be able to chip the tensors, which in turn requires us to have buffer views. I don't want to work on those until we have the type-erased floating-point. For now I'll just do the DFT contraction with a custom kernel.

@ryanmrichard ryanmrichard merged commit 9481ba1 into master Sep 26, 2025
7 checks passed
@ryanmrichard ryanmrichard deleted the mixed_contraction branch September 26, 2025 02:56
@github-actions
Copy link
Contributor

🚀 [bumpr] Bumped!
New version:v0.0.67
Changes:v0.0.66...v0.0.67

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants